import os
import sys

# 将项目根目录添加到Python路径
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from app import create_app
from models import User, db

def create_admin():
    """创建管理员账户"""
    try:
        app = create_app()
        with app.app_context():
            # 检查管理员是否已存在
            admin = User.query.filter_by(username='admin').first()
            if admin:
                print("管理员账户已存在！")
                return
            
            # 创建管理员账户
            admin = User(
                username='admin',
                email='admin@example.com',
                role='admin',  # 设置角色为管理员
                name='系统管理员'
            )
            admin.set_password('admin123')  # 设置初始密码
            
            db.session.add(admin)
            db.session.commit()
            
            print("管理员账户创建成功！")
            print("用户名: admin")
            print("密码: admin123")
            print("请登录后立即修改密码！")
            
    except Exception as e:
        print(f"创建失败：{str(e)}")
        sys.exit(1)

if __name__ == '__main__':
    create_admin() 